
class CNNConfig:
    def __init__(self, batch_size, pre_vector = True):
        self.batch_size = 16
        self.word_vec_num = 300
        self.seq_len = 720
        self.dropout = 0.5
        self.class_num = 8
        self.pre_word_vec = pre_vector

class RNNConfig:
    def __init__(self, batch_size, pre_vector = True):
        self.batch_size = 16
        self.word_vec_num = 300
        self.dropout = 0.5
        self.hidden_num = 128
        self.class_num = 8
        self.pre_word_vec = pre_vector
        self.attn_hidden_num = 128

class MLPConfig:
    def __init__(self, batch_size, pre_vector = True):
        self.batch_size = batch_size
        self.word_vec_num = 300
        self.pre_word_vec = pre_vector
        self.dropout = 0.5
        self.attn_hidden_num = 128
        self.hidden_size = [300, 256, 256, 128, 64]
        self.class_num = 8